# coding:utf-8
# 曾山
# 时间 2022/5/4 18:16
from flask import g

'''
自定义一个装饰器，验证某些请求是否已经登录过了  如果已经登录了继续访问
'''
def login_required(func):
    def wrapper(*args,**kwargs):
        if g.user_id is not None:
            return func(*args,**kwargs)
        else:
            return {'message':' Invalid token.'},401

    return wrapper


